<template>
  <Tree v-bind="getAttrs" v-model:selectedKeys="state">
    <template #[item]="data" v-for="item in Object.keys($slots)">
      <slot :name="item" v-bind="data || {}"></slot>
    </template>
  </Tree>
</template>

<script lang="ts" setup>
import { type Recordable, type AnyFunction } from "@vben/types";
import {
  type PropType,
  computed,
  watch,
  ref,
  onMounted,
  unref,
  useAttrs,
} from "vue";
import { Tree, TreeProps } from "ant-design-vue";
import { isArray, isFunction } from "@/utils/is";
import { get } from "lodash-es";
import { DataNode } from "ant-design-vue/es/tree";
import { useRuleFormItem } from "@/hooks/component/useFormItem";

defineOptions({ name: "ApiTree" });

const props = defineProps({
  api: { type: Function as PropType<(arg?: any) => Promise<Recordable<any>>> },
  params: { type: Object },
  immediate: { type: Boolean, default: true },
  resultField: { type: String, default: "" },
  afterFetch: { type: Function as PropType<AnyFunction> },
  value: {
    type: Array as PropType<TreeProps["selectedKeys"]>,
  },
});

const emit = defineEmits(["options-change", "change", "update:value"]);

const attrs = useAttrs();

const treeData = ref<DataNode[]>([]);
const isFirstLoaded = ref<Boolean>(false);
const loading = ref(false);
const emitData = ref<any[]>([]);

const [state] = useRuleFormItem(props, "value", "change", emitData);
const getAttrs = computed(() => {
  return {
    ...(props.api ? { treeData: unref(treeData) } : {}),
    ...attrs,
  };
});

watch(
  () => state.value,
  (v) => {
    emit("update:value", v);
  },
);

watch(
  () => props.params,
  () => {
    !unref(isFirstLoaded) && fetch();
  },
  { deep: true },
);

watch(
  () => props.immediate,
  (v) => {
    v && !isFirstLoaded.value && fetch();
  },
);

onMounted(() => {
  props.immediate && fetch();
});

async function fetch() {
  const { api, afterFetch } = props;
  if (!api || !isFunction(api)) return;
  loading.value = true;
  treeData.value = [];
  let result;
  try {
    result = await api(props.params);
  } catch (e) {
    console.error(e);
  }
  if (afterFetch && isFunction(afterFetch)) {
    result = afterFetch(result);
  }
  loading.value = false;
  if (!result) return;
  if (!isArray(result)) {
    result = get(result, props.resultField);
  }
  treeData.value = (result as (Recordable & { key: string | number })[]) || [];
  isFirstLoaded.value = true;
  emit("options-change", treeData.value);
}
</script>
